"""
一条包含字母 A-Z 的消息通过以下方式进行了编码：

'A' -> 1
'B' -> 2
...
'Z' -> 26
给定一个只包含数字的非空字符串，请计算解码方法的总数。

示例 1:

输入: "12"
输出: 2
解释: 它可以解码为 "AB"（1 2）或者 "L"（12）。
示例 2:

输入: "226"
输出: 3
解释: 它可以解码为 "BZ" (2 26), "VF" (22 6), 或者 "BBF" (2 2 6) 。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/decode-ways
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
"""


"""
一共提交了5次， 有毒啊

如果i的值和 [i-1]在10-26之间
    如果 [i] != 0
        dp[i] = dp[i-1] + dp[i-2]
    else:
        dp[i] = dp[i-2]

如果不在，且[i] == 0:
    表面这个表达式是错误的，直接返回0

"""
# 26.76%
class Solution:
    def numDecodings(self, s: str) -> int:

        if not(s):
            return 0
        if s[0] == "0":
            return 0
        dp = [1] * len(s)
        dp[0] = 1
        for i in range(1,len(s)):
            if 10 <= int(s[i-1:i+1]) <= 26:
                if s[i] == "0":
                    dp[i] = dp[i-2]
                else:
                    dp[i] = dp[i-1] + dp[i-2]
            else:
                if s[i] == "0":
                    return 0
                dp[i] = dp[i-1]
        print(dp)
        return dp[-1]

def test():
    assert Solution().numDecodings("10") == 1
    assert Solution().numDecodings("17") == 2
    assert Solution().numDecodings("101") == 1
    assert Solution().numDecodings("100") == 0
    assert Solution().numDecodings("120") == 1
    assert Solution().numDecodings("1272") == 2
    assert Solution().numDecodings("1262") == 3
    assert Solution().numDecodings("1212") == 5
    assert Solution().numDecodings("12120") == 3
    assert Solution().numDecodings("12020") == 1
    assert Solution().numDecodings("11111111") == 34


if __name__ == "__main__":
    print(Solution().numDecodings("226120"))
